#!/bin/bash

# Visitor-Counter-Cheater 浏览器安装脚本
# 自动检测系统并安装 Chrome 或 Firefox

echo "🚀 开始安装浏览器..."

# 检测系统类型
if [ -f /etc/os-release ]; then
    . /etc/os-release
    OS=$NAME
    VER=$VERSION_ID
else
    echo "❌ 无法检测操作系统类型"
    exit 1
fi

echo "📋 检测到系统: $OS $VER"

# 检查是否已安装浏览器
check_browser() {
    echo "🔍 检查已安装的浏览器..."
    
    if command -v google-chrome-stable &> /dev/null || \
       command -v google-chrome &> /dev/null || \
       command -v chromium-browser &> /dev/null || \
       command -v chromium &> /dev/null; then
        echo "✅ Chrome/Chromium 已安装"
        return 0
    fi
    
    if command -v firefox &> /dev/null || command -v firefox-esr &> /dev/null; then
        echo "✅ Firefox 已安装"
        return 0
    fi
    
    echo "❌ 未找到浏览器"
    return 1
}

# 安装浏览器
install_browser() {
    case $OS in
        *Ubuntu*|*Debian*)
            echo "📦 使用 apt 安装浏览器..."
            sudo apt update
            
            # 尝试安装 Chromium
            if sudo apt install -y chromium-browser; then
                echo "✅ Chromium 安装成功"
            elif sudo apt install -y chromium; then
                echo "✅ Chromium 安装成功"
            else
                # 安装 Firefox
                sudo apt install -y firefox
                echo "✅ Firefox 安装成功"
            fi
            ;;
        *CentOS*|*Red*|*Fedora*)
            echo "📦 使用 yum/dnf 安装浏览器..."
            
            if command -v dnf &> /dev/null; then
                PKG_MGR="dnf"
            else
                PKG_MGR="yum"
            fi
            
            sudo $PKG_MGR update -y
            
            # 尝试安装 Chromium
            if sudo $PKG_MGR install -y chromium; then
                echo "✅ Chromium 安装成功"
            else
                # 安装 Firefox
                sudo $PKG_MGR install -y firefox
                echo "✅ Firefox 安装成功"
            fi
            ;;
        *)
            echo "❌ 不支持的操作系统: $OS"
            echo "💡 请手动安装以下任一浏览器："
            echo "   - Chrome/Chromium"
            echo "   - Firefox"
            exit 1
            ;;
    esac
}

# 主流程
if check_browser; then
    echo "🎉 浏览器已就绪，无需安装"
else
    install_browser
    
    # 验证安装
    if check_browser; then
        echo "🎉 浏览器安装完成！"
    else
        echo "❌ 浏览器安装失败，请手动安装"
        exit 1
    fi
fi

echo ""
echo "📋 下一步："
echo "1. 运行: pip install selenium webdriver-manager"
echo "2. 运行: python3 selenium_all_website.py"
echo ""
echo "💡 提示：脚本已自动支持 Chrome 和 Firefox 浏览器"